/*
 * This file is part of aion-unique <aion-unique.org>.
 *
 *  aion-unique is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  aion-unique is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with aion-unique.  If not, see <http://www.gnu.org/licenses/>.
 */

package com.aionengine.gameserver.dataholders;

import com.aionengine.gameserver.model.templates.chest.ChestTemplate;
import gnu.trove.map.hash.THashMap;
import gnu.trove.map.hash.TIntObjectHashMap;

import javax.xml.bind.Unmarshaller;
import javax.xml.bind.annotation.XmlAccessType;
import javax.xml.bind.annotation.XmlAccessorType;
import javax.xml.bind.annotation.XmlElement;
import javax.xml.bind.annotation.XmlRootElement;
import java.util.ArrayList;
import java.util.List;

/**
 * @author Wakizashi
 */
@XmlRootElement(name = "chest_templates")
@XmlAccessorType(XmlAccessType.FIELD)
public class ChestData {

    @XmlElement(name = "chest")
    private List<ChestTemplate> chests;

    /**
     * A map containing all npc templates
     */
    private TIntObjectHashMap<ChestTemplate> chestData = new TIntObjectHashMap<ChestTemplate>();
    private TIntObjectHashMap<ArrayList<ChestTemplate>> instancesMap = new TIntObjectHashMap<ArrayList<ChestTemplate>>();
    private THashMap<String, ChestTemplate> namedChests = new THashMap<String, ChestTemplate>();

    /**
     * - Inititialize all maps for subsequent use - Don't nullify initial chest list as it will be used during reload
     *
     * @param u
     * @param parent
     */
    void afterUnmarshal(Unmarshaller u, Object parent) {
        chestData.clear();
        instancesMap.clear();
        namedChests.clear();

        for (ChestTemplate chest : chests) {
            chestData.put(chest.getNpcId(), chest);
            if (chest.getName() != null && !chest.getName().isEmpty())
                namedChests.put(chest.getName(), chest);
        }
    }

    public int size() {
        return chestData.size();
    }

    /**
     * @param npcId
     * @return
     */
    public ChestTemplate getChestTemplate(int npcId) {
        return chestData.get(npcId);
    }

    /**
     * @return the chests
     */
    public List<ChestTemplate> getChests() {
        return chests;
    }

    /**
     * @param chests the chests to set
     */
    public void setChests(List<ChestTemplate> chests) {
        this.chests = chests;
        afterUnmarshal(null, null);
    }
}
